package org.tity.commons.json;

import java.util.List;

import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;

/**
 * @author fangjieli
 */
public class JSONUtil {
 
	/**
	 * 格式化 json
	 * @param jsonstr 输入的JSON字符串
	 * @return 格式化后的JSON字符串
	 * */
	public static String format(String jsonstr)
	{
		return JSONFormater.format(jsonstr);
	}
	
	/**
	 * 格式化 json
	 * @param json 输入的JSON对象
	 * @return 格式化后的JSON字符串
	 * */
	public static String format(JSONObject json)
	{
		return JSONFormater.format(json.toJSONString());
	}
	
	/**
	 * 格式化 json
	 * @param json 输入的JSON数组
	 * @return 格式化后的JSON字符串
	 * */
	public static String format(JSONArray json)
	{
		return JSONFormater.format(json.toJSONString());
	}
	
	
    /**
     * xml转json
     * @param xml xml字符串
     * @return Json对象
     * @throws DocumentException 异常
     */
    public static JSONObject toJSONObject(String xml) throws DocumentException{
        Document doc= DocumentHelper.parseText(xml);
        return toJSONObject(doc);
    }


    /**
     * xml转json
     * @param doc  xml文档
     * @return Json对象
     */
    public static JSONObject toJSONObject(Document doc) {
		JSONObject json=new JSONObject(true);
        dom2JSON(doc.getRootElement(), json);
        return json;
	}
    
     

    /**
     * xml转json
     * @param element
     * @param json
     */
    private static void dom2JSON(Element element,JSONObject json){
        //如果是属性
        for(Object o:element.attributes()){
            Attribute attr=(Attribute)o;
            if(!isEmpty(attr.getValue())){
                json.put("@"+attr.getName(), attr.getValue());
            }
        }
        List<Element> chdEl=element.elements();
        if(chdEl.isEmpty()&&!isEmpty(element.getText())){//如果没有子元素,只有一个值
            json.put(element.getName(), element.getText());
        }

        for(Element e:chdEl){//有子元素
            if(!e.elements().isEmpty()){//子元素也有子元素
                JSONObject chdjson=new JSONObject();
                dom2JSON(e,chdjson);
                Object o=json.get(e.getName());
                if(o!=null){
                    JSONArray jsona=null;
                    if(o instanceof JSONObject){//如果此元素已存在,则转为jsonArray
                        JSONObject jsono=(JSONObject)o;
                        json.remove(e.getName());
                        jsona=new JSONArray();
                        jsona.add(jsono);
                        jsona.add(chdjson);
                    }
                    if(o instanceof JSONArray){
                        jsona=(JSONArray)o;
                        jsona.add(chdjson);
                    }
                    json.put(e.getName(), jsona);
                }else{
                    if(!chdjson.isEmpty()){
                        json.put(e.getName(), chdjson);
                    }
                }


            }else{//子元素没有子元素
                for(Object o:element.attributes()){
                    Attribute attr=(Attribute)o;
                    if(!isEmpty(attr.getValue())){
                        json.put("@"+attr.getName(), attr.getValue());
                    }
                }
                if(!e.getText().isEmpty()){
                    json.put(e.getName(), e.getText());
                }
            }
        }
    }

    private static boolean isEmpty(String str) {
        if (str == null || str.trim().isEmpty() || "null".equals(str)) {
            return true;
        }
        return false;
    }
}
